how to avoid return none in python